import numpy as np

A = np.arange(3, 15)
print('A = ', A)
print(A[3])

A = A.reshape(3, 4)
print('A = ', A)
print(A[2])
print(A[2][2])
print(A[2, 2])

# 也可以使用冒号表达式
# 第3行的所有数
print(A[2, :])
# 第3列的所有数
print(A[:, 2])

# 按行遍历矩阵
for row in A:
    print(row)

# 按列遍历矩阵
for col in A.T:
    print(col)

# 二维矩阵变成一维(展开)
print(A.flatten())